tensor([[[0.2900, 0.2900],
         [0.2900, 0.2900]],

        [[0.2900, 0.2900],
         [0.2900, 0.2900]],

        [[0.4200, 0.4200],
         [0.4200, 0.4200]]], grad_fn=<RoundBackward1>)